﻿#include<iostream>
#include<vector>
#include<algorithm>
//寻找三数之和
/*
*给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，
*同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。
*注意：答案中不可以包含重复的三元组。
*/
using namespace std;
class Solution1 {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ret;
        sort(nums.begin(), nums.end());
        for (int i = nums.size() - 1; i >= 0;) {
            int target = -nums[i];
            int left = 0, right = i - 1;
            while (left < right) {
                int sum = nums[left] + nums[right];
                if (sum > target) right--;
                else if (sum < target) left++;
                else {
                    ret.push_back({ nums[left],nums[right],nums[i] });
                    left++, right--;
                    while (left < right && nums[left] == nums[left - 1]) left++;
                    while (left < right && nums[right] == nums[right + 1]) right--;//这题要注意要求不能有重复的三元组，所以要消除重复
                }
            }
            i--;
            while (i > 0 && nums[i] == nums[i + 1]) i--;//消除重复
        }
        return ret;
    }
};
//四数之和
/*
*给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且
*不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）
*/
class Solution2 {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ret;
        sort(nums.begin(), nums.end());
        for (int i = nums.size() - 1; i >= 0;) {
            for (int j = i - 1; j >= 0;) {
                long long k = (long long)target - nums[i] - nums[j];
                int left = 0, right = j - 1;
                while (left < right) {
                    int sum = nums[left] + nums[right];
                    if (sum > k) right--;
                    else if (sum < k) left++;
                    else {
                        ret.push_back({ nums[left],nums[right],nums[j],nums[i] });
                        left++, right++;
                        while (left < right && nums[left] == nums[left - 1]) left++;
                        while (left < right && nums[right] == nums[right + 1]) right--;
                    }
                }
                j--;
                while (j > 0 && nums[j] == nums[j + 1])j--;
            }
            i--;
            while (i > 0 && nums[i] == nums[i + 1])i--;
        }
        return ret;
    }
};